set(EXAMPLE_DIR ${CMAKE_CURRENT_SOURCE_DIR})

set(CNN ON)
set(ResNet ON)
set(RNN ON)

add_subdirectory(src)
include_directories(include ${ROOT_DIR}/include)

if (OpenCV_FOUND)
    include_directories(${OpenCV_INCLUDE_DIRS})
endif()

if (CNN AND OpenCV_FOUND)
    add_executable(cnn_demo CNN.cpp)
    target_link_libraries(cnn_demo ${OpenCV_LIBRARIES} DATASET MINNET)
endif()

if (ResNet AND OpenCV_FOUND)
    add_executable(resnet_demo ResNet.cpp)
    target_link_libraries(resnet_demo ${OpenCV_LIBRARIES} DATASET MINNET)
endif()

if(RNN)
    add_executable(rnn_demo RNN.cpp)
    target_link_libraries(rnn_demo DATASET MINNET)
endif()